/*
 * Copyright 2002-2025 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.test.context.bean.override.mockito;

import java.util.ArrayList;
import java.util.List;

import org.mockito.Mockito;

/**
 * Beans created using Mockito.
 *
 * @author Andy Wilkinson
 * @author Sam Brannen
 * @since 6.2
 */
class MockBeans {

	private final List<Object> beans = new ArrayList<>();


	void add(Object bean) {
		this.beans.add(bean);
	}

	/**
	 * Reset all Mockito beans configured with the supplied {@link MockReset} strategy.
	 * <p>No mocks will be reset if the supplied strategy is {@link MockReset#NONE}.
	 */
	void resetAll(MockReset reset) {
		if (reset != MockReset.NONE) {
			for (Object bean : this.beans) {
				if (reset == MockReset.get(bean)) {
					Mockito.reset(bean);
				}
			}
		}
	}

}
